package com.chinaisafe.gateway.config;

import cn.dev33.satoken.dao.SaTokenDao;
import cn.dev33.satoken.listener.SaTokenListener;
import cn.dev33.satoken.stp.SaLoginModel;
import cn.hutool.core.util.StrUtil;
import com.chinaisafe.gateway.sdk.constant.GateWayConstant;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;


/**
 * GatewaySaTokenListener
 * Sa-Token 时间监听改写
 * @author xuefu.shu
 * @date 2024-07-09 10:42
 */
@Slf4j
@Component
public class GatewaySaTokenListener implements SaTokenListener {

    @Autowired
    private SaTokenDao saTokenDao;

    /** 每次登录时触发 */
    @Override
    public void doLogin(String loginType, Object loginId, String tokenValue, SaLoginModel loginModel) {
        if (log.isDebugEnabled()) {
            log.debug("---------- 自定义侦听器实现 doLogin");
        }
    }

    /** 每次注销时触发 */
    @Override
    public void doLogout(String loginType, Object loginId, String tokenValue) {
        if (log.isDebugEnabled()) {
            log.info("---------- 自定义侦听器实现 doLogout");
        }
        saTokenDao.delete(StrUtil.format(GateWayConstant.TENANCY_REDIS_KEY, tokenValue));
    }

    /** 每次被踢下线时触发 */
    @Override
    public void doKickout(String loginType, Object loginId, String tokenValue) {
        if (log.isDebugEnabled()) {
            log.info("---------- 自定义侦听器实现 doKickout");
        }
        saTokenDao.delete(StrUtil.format(GateWayConstant.TENANCY_REDIS_KEY, tokenValue));
    }

    /** 每次被顶下线时触发 */
    @Override
    public void doReplaced(String loginType, Object loginId, String tokenValue) {
        if (log.isDebugEnabled()) {
            log.info("---------- 自定义侦听器实现 doReplaced");
        }
        saTokenDao.delete(StrUtil.format(GateWayConstant.TENANCY_REDIS_KEY, tokenValue));
    }

    /** 每次被封禁时触发 */
    @Override
    public void doDisable(String loginType, Object loginId, String service, int level, long disableTime) {
        if (log.isDebugEnabled()) {
            log.info("---------- 自定义侦听器实现 doDisable");
        }
    }

    /** 每次被解封时触发 */
    @Override
    public void doUntieDisable(String loginType, Object loginId, String service) {
        if (log.isDebugEnabled()) {
            log.info("---------- 自定义侦听器实现 doUntieDisable");
        }
    }

    /** 每次二级认证时触发 */
    @Override
    public void doOpenSafe(String loginType, String tokenValue, String service, long safeTime) {
        if (log.isDebugEnabled()) {
            log.info("---------- 自定义侦听器实现 doOpenSafe");
        }
    }

    /** 每次退出二级认证时触发 */
    @Override
    public void doCloseSafe(String loginType, String tokenValue, String service) {
        if (log.isDebugEnabled()) {
            log.info("---------- 自定义侦听器实现 doCloseSafe");
        }
    }

    /** 每次创建Session时触发 */
    @Override
    public void doCreateSession(String id) {
        if (log.isDebugEnabled()) {
            log.info("---------- 自定义侦听器实现 doCreateSession");
        }
    }

    /** 每次注销Session时触发 */
    @Override
    public void doLogoutSession(String id) {
        if (log.isDebugEnabled()) {
            log.info("---------- 自定义侦听器实现 doLogoutSession");
        }
    }
    
    /** 每次Token续期时触发 */
    @Override
    public void doRenewTimeout(String tokenValue, Object loginId, long timeout) {
        if (log.isDebugEnabled()) {
            log.info("---------- 自定义侦听器实现 doRenewTimeout");
        }
        saTokenDao.updateTimeout(StrUtil.format(GateWayConstant.TENANCY_REDIS_KEY, tokenValue), timeout);
    }
}
